from sqlalchemy.orm import sessionmaker,scoped_session
from engine import connect
from books import Books
#执行数据库操作需要创建session
def session(conn):
    Session = sessionmaker(bind=conn) #每次操作需要创建session
    session = Session()
    #线程安全，本地线程实现每个线程用同一个session
    session = scoped_session(Session)
    return session
def insert(session): #添加记录
    books = []
    while 1:
        title = input("输入书名(0=退出):")
        if title == "0":
            break
        book = Books(书名 = title,作者 = input("输入作者："),出版社=input("输入出版社:"),
                     出版日期=input("输入出版日期:"),定价=input("输入定价:"))
        books.append(book)
    session.add_all(books) #批量添加数据
    session.commit() #提交添加的数据
    session.close() #关闭session
    print("所有记录添加成功!")
if __name__ == "__main__":
    conn = connect() #创建数据库连接
    session = session(conn) #创建线程
    insert(session) #调用添加记录方法
